package logic

import (
	"context"
	"fmt"
	"meituan-waimai/models/cache"
	"meituan-waimai/models/mysql"
	"meituan-waimai/pkg"

	"meituan-waimai/meituanrpc/internal/svc"
	"meituan-waimai/meituanrpc/meituanrpc"

	"github.com/zeromicro/go-zero/core/logx"
)

type LoginLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic {
	return &LoginLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *LoginLogic) Login(in *meituanrpc.LoginRequest) (*meituanrpc.LoginResponse, error) {
	// todo: add your logic here and delete this line
	//参数验证
	if in.Mobile == "" && in.Account == "" {
		return nil, fmt.Errorf("参数不可为空")
	}

	var user mysql.User
	// 如果是手机号登录
	if in.Mobile != "" {
		// 判断手机号注册的用户是否存在
		mysql.DB.Where("mobile = ?", in.Mobile).First(&user)
		if user.ID == 0 {
			return nil, fmt.Errorf("手机号未注册，请前往注册")
		}
		// 如果存在则进行验证码验证
		smsCode := cache.RDB.Get(l.ctx, "SmsCode:"+in.Mobile).Val()
		if smsCode != in.SmsCode {
			return nil, fmt.Errorf("短信验证码错误，请重新输入")
		}
	}

	// 如果是用户名登录
	if in.Account != "" {
		mysql.DB.Where("account = ?", in.Account).First(&user)
		if user.ID == 0 {
			return nil, fmt.Errorf("账号未注册，请前往注册")
		}
		if user.Password != pkg.Md5(in.Password) {
			return nil, fmt.Errorf("密码输入错误，请重新输入")
		}
	}

	return &meituanrpc.LoginResponse{
		UserId: int64(user.ID),
	}, nil
}
